Skip to content

ENH: test count_nonzero with tuple axes#372

Merged
ev-br merged 1 commit intodata-apis:masterfrom
ev-br:count_nonzero_tuples
May 9, 2025
Merged

ENH: test count_nonzero with tuple axes#372
ev-br merged 1 commit intodata-apis:masterfrom
ev-br:count_nonzero_tuples

Conversation

@ev-br
Copy link
Member

@ev-br ev-br commented May 8, 2025

array-api-compat had a bug in count_nonzero with tuple axes and keepdims=True on torch. This buggy code is in array_api_compat because torch.count_nonzero does not have keepdims and array_api_compat has to emulate it.
The fix is in data-apis/array-api-compat#318

@ev-br ev-br merged commit 66e8cad into data-apis:master May 9, 2025
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant